package com.ting.test.algorithms.并查集.岛问题;

/**
 * 岛问题
 * <p>
 * 给定一个二维数组matrix，里面的值不是1就是0，
 * 上、下、左、右相邻的1认为是一片岛，
 * 返回matrix中岛的数量
 * 思路，只要发现一个'1' 就把他上下左右位置的1都感染了
 * 从左到右，从上到下开始循环x判断：
 * 1. 如果当前位置 d[i][j]==1,则认为是一个新的岛，
 * 2. 将当前位置置为2，表示已经被感染（处理）
 * 3. 依次递归感染当前位置的上下左右四个位置。
 */
public class 递归解法 {
    public static void main(String[] args) {
//        int[][] arr = new int[][]{{1,1,1,1,1,1,0,0,0,0,0},{1,1,1,1,1,1,0,1,0,1,0},{1,1,1,1,1,1,0,0,0,0,0}};
        int[][] arr = new int[][]{{1,1,1},{1,0,1},{1,1,1}};
        int count = getIslandNums(arr);
        System.out.println(count);
    }

    private static int getIslandNums(int[][] arr) {
        int count = 0;
        for (int i = 0; i < arr.length; i++) {
            for (int j = 0; j < arr[0].length; j++) {
                if (arr[i][j] == 1) {
                    count++;
                    infect(arr, i, j);
                }
            }
        }
        return count;
    }


    private static void infect(int[][] arr, int i, int j) {
        if (i < 0 || i >= arr.length || j < 0 || j >= arr[0].length|| arr[i][j]!=1) {
            return;
        }
        if (arr[i][j] == 1) {
            arr[i][j] = 2;
        }
        infect(arr, i - 1, j);
        infect(arr, i + 1, j);
        infect(arr, i, j - 1);
        infect(arr, i, j + 1);
    }
}
